import sys
sys.path.append("/home/colector1337/Programming/ZPR/bin/gcc-4.6/debug/")
import ZPRfaif

print 'Now running tests from test.py..'

DecisionTree=ZPRfaif.ZPRDecisionTree()
DecisionTree.zprPassDomain("outlook", ["sunny", "overcast", "rain"])
DecisionTree.zprPassDomain("temperature", ["hot", "mild", "cold"])
DecisionTree.zprPassDomain("humidity",["normal", "high"])
DecisionTree.zprPassDomain("wind",["strong", "weak"])
DecisionTree.zprPassAttributeDomain(["good", "bad"])

DecisionTree.applyTree()

DecisionTree.zprPassTrainExamples([ "sunny", "hot", "high", "weak"] , "bad")
DecisionTree.zprPassTrainExamples([ "sunny", "hot", "high", "strong"], "bad" )
DecisionTree.zprPassTrainExamples([ "overcast", "hot", "high", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "rain", "mild", "high", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "rain", "cold", "normal", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "rain", "cold", "normal", "strong"], "bad" )
DecisionTree.zprPassTrainExamples([ "overcast", "cold", "normal", "strong"], "good" )
DecisionTree.zprPassTrainExamples([ "sunny", "mild", "high", "weak"], "bad" )
DecisionTree.zprPassTrainExamples(["sunny", "cold", "normal", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "rain", "mild", "normal", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "sunny", "mild", "normal", "strong"], "good" )
DecisionTree.zprPassTrainExamples([ "overcast", "mild", "high", "strong"], "good" )
DecisionTree.zprPassTrainExamples([ "overcast", "hot", "normal", "weak"], "good" )
DecisionTree.zprPassTrainExamples([ "rain", "mild", "high", "strong"], "bad" )

DecisionTree.Train()

DecisionTree.zprPassTestExamples(["overcast", "hot", "high", "weak"])

Domain = DecisionTree.zprGetCategoryDomainFromDTC()

AttrDomains = DecisionTree.zprGetAttrDomainsFromDTC()

Category = DecisionTree.zprGetCategoryFromDTC(["overcast", "hot", "high", "weak"])

CategoryIdd = DecisionTree.zprGetCategoryIddFromDTC("good")

Categories = DecisionTree.zprGetCategoriesFromDTC(["overcast", "hot", "high", "weak"])

trainParam = ZPRfaif.DecisionTreeTrainParam()
